package stmo.链表;

import java.util.Iterator;

/**
 *  双向链表
 */


public class TowWayLinkList<T> implements Iterable<T> {

    //首结点
    private Node head;

    //最后一个结点
    private Node last;

    //链表的长度
    private int N;

    private class Node {

        //存储的数据
        public T item;

        //指向上一个结点
        public Node pre;

        //指向下一个结点
        public Node next;

        public Node(T item,Node pre,Node next) {

            this.item = item;
            this.pre = pre;
            this.next = next;
        }
    }

    //构造方法
    public TowWayLinkList() {
        //初始化头结点和尾节点
        this.head = new Node(null,null,null);
        //初始化元素个数
        this.N = 0;
    }

    //清空链表
    public void clear() {
        this.head.next = null;
        this.last.item = null;
        this.last.pre = null;
        this.N = 0;
    }

    //获取链表的长度
    public int length() {
        return N;
    }

    //判断链表是否为空
    public boolean isEmpty() {
        return N == 0;
    }

    //获取第一个元素
    public T getFirst() {
        if(isEmpty()) {
            return null;
        }
        return head.next.item;
    }

    //获取最后一个元素
    public T getLast() {
        if(isEmpty()) {
            return null;
        }
        return last.item;
    }

    //向链表中添加元素t
    public void insert(T t) {
        //如果链表为空
        if(isEmpty()) {
            //创建新的结点
            Node n = new Node(t,head,null);
            //让新结点成为尾结点
            last = n;
            //让头结点指向新结点
            head.next = n;
        }else {
            //如果链表不为空
            //创建新的结点,让新结点的pro指向前尾结点
            Node oldLast = last;
            Node n = new Node(t,oldLast,null);
            //让当前尾结点指向新结点
            oldLast.next = n;
            //让新结点成为尾结点
            this.last = n;
        }
        //元素个数+1
        N++;
    }

    //向指定位置i处，添加元素t
    public void insert(int i,T t) {
        //找到i位置的前一个结点
        Node pre = head;
        for (int index = 0;index <= i-1;index++) {
            pre = pre.next;
        }
        //找到i位置的结点
        Node curr = pre.next;
        //创建新结点
        Node addNode = new Node(t,pre,curr);
        //让i位置的前一个结点的下一个结点变为新结点
        pre.next = addNode;
        //让i位置的前一个结点变为新结点
        curr.pre = addNode;
        //元素个数+1
        N++;
    }

    //获取指定位置i处的元素
    public T get(int i) {
        Node n = head.next;
        for (int index = 0;index < i;index++) {
            n = n.next;
        }
        return n.item;
    }


    //删除指定位置i处的元素，并返回被删除的元素
    public T remove(int i) {
        //找到i位置的前一个结点
        Node pre = head;
        for (int index = 0;index <= i-1;index++) {
            pre = pre.next;
        }
        //找到i位置的结点
        Node currNode = pre.next;
        if (currNode.next == null) {
            pre.next = null;
            last = pre;
            N--;
            return currNode.item;
        }
        //找到i位置的下一个结点
        Node nextNode = currNode.next;
        //让i位置的前一个结点的下一个结点变为i位置的下一个结点
        pre.next = nextNode;
        //让i位置的下一个结点的上一个结点变为i位置的前一个结点
        nextNode.pre = pre;
        //元素个数-1
        N--;
        return currNode.item;
    }

    //查找元素t在链表中第一次出现的位置
    public int indexOf(T t) {
        Node n = head;
        for (int i = 0;n.next != null;i++) {
            n = n.next;
            if (n.item.equals(t)) {
                return i;
            }
        }
        return -1;
    }

    /**
     *  遍历
     */
    @Override
    public Iterator<T> iterator() {
        return new TIterator();
    }

    private class TIterator implements Iterator {

        private Node n;

        public TIterator() {
            this.n = head;
        }

        @Override
        public boolean hasNext() {
            return n.next != null;
        }

        @Override
        public Object next() {
            n = n.next;
            return n.item;
        }
    }

}
